// Thread.hpp 线程封装
#ifndef _THREAD_H_
#define _THREAD_H_

#include <iostream>
#include <string>
#include <functional>
#include <pthread.h>
#include <cstring>
#include <cstdio>
#include "Log.hpp"

namespace ThreadModlue
{
    using namespace LogMoudle;
    // 原子计数器 —— bug
    static uint32_t number = 1;

    class Thread
    {
        // 线程要执行的外部方法，我们不考虑传参，后续有std::bind来进行类间耦合
        using func_t = std::function<void()>;

    private:
        void EnableDetach() // 启用分离操作
        {
            // std::cout << _name << " 线程被分离了" << std::endl;
            _isdetach = true;
        }
        void EnableRunning()
        {
            _isrunning = true;
        }
        
        // Routine属于类内的成员函数，默认传参this指针
        // 设置static就不传this指针了
        static void *Routine(void *args)
        {
            Thread *self = static_cast<Thread *>(args);
            self->EnableRunning();
            if (self->_isdetach)
                self->Detach(); // 分离线程
            pthread_setname_np(self->_tid, self->_name.c_str()); // 为线程设计名字
            self->_func();      // 回调处理

            return nullptr;
        }

    public:
        Thread(func_t func)
            : _tid(0),
              _isdetach(false), // 默认不是分离状态
              _isrunning(false),
              _res(nullptr),
              _func(func)
        {
            _name = "thread-" + std::to_string(number++);
        }

        void Detach()
        {
            if (_isdetach)
                return;
            if (_isrunning) // 运行中调用detach进行分离
                pthread_detach(_tid);
            EnableDetach(); // 如果没有运行直接对标志位置1
        }

        bool Start()
        {
            if (_isrunning)
                return false;
            int n = pthread_create(&_tid, nullptr, Routine, this); // 传入this指针实现回调机制
            if (n != 0)
            {
                std::cerr << "Create thread error!" << std::endl;
                return false;
            }
            else
            {
                std::cout << _name << " create success!" << std::endl;
            }
            return true;
        }

        bool Stop()
        {
            if (_isrunning)
            {
                int n = pthread_cancel(_tid);
                if (n != 0)
                {
                    // std::cerr << "Cancel thread error!" << std::endl;
                    return false;
                }
                else
                {
                    _isrunning = false;
                    // std::cout << _name << " stop success!" << std::endl;
                }
            }
            return true;
        }

        void Join()
        {
            if (_isdetach)
            {
                // std::cout << _name << " 线程已经是分离得了, 不能进行join!" << std::endl;
                return;
            }
            int n = pthread_join(_tid, &_res);
            if (n != 0)
            {
                // std::cerr << "Join thread error!" << std::endl;
                LOG(LogLevel::DEBUG) << "Join线程失败!";
            }
            else
            {
                // std::cout << _name << " join success!" << std::endl;
                LOG(LogLevel::DEBUG) << "Join线程成功!";
            }
        }

        std::string Name()
        {
            return _name;
        }

        ~Thread()
        {
        }

    private:
        pthread_t _tid;
        std::string _name;
        bool _isdetach;
        bool _isrunning;
        void *_res;
        func_t _func;
    };
}

#endif